# 自定义filter的基本语法
from django import template
from ..models import SecondLevelTab

''' 
# 将自定义的 filter 注册到程序中
@register.filter
@register.filter(name = 'fleter_name') 这个装饰器把自定义的filter注册到程序中，
让其可以使用自定义的 filter 方法； reg是 ‘|’ 前面的变量 reg2是方法后面的参数。
'''

register = template.Library() # 实例化一个模板对象

@register.filter(name = 'getSecondTab')
def getSecondTab(tab_id):
    result = SecondLevelTab.objects.filter(tab_id=tab_id)
    if list(result) == []:
        return False
    else:
        return result

